//
/*
给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和

叉搜索树满足下列约束条件：
    节点的左子树仅包含键 小于 节点键的节点。
    节点的右子树仅包含键 大于 节点键的节点。
    左右子树也必须是二叉搜索树

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：

输入：root = [0,null,1]
输出：[1,null,1]

示例 3：

输入：root = [1,0,2]
输出：[3,3,2]

示例 4：

输入：root = [3,2,4,1]
输出：[7,9,4,10]


*/

//反序中序遍历
class Solution
{
public:
    int sum = 0;

    TreeNode *convertBST(TreeNode *root)
    {
        if (root != nullptr)
        {
            convertBST(root->right);
            sum += root->val;
            root->val = sum;
            convertBST(root->left);
        }
        return root;
    }
};

//Morris 遍历
class Solution
{
public:
    TreeNode *getSuccessor(TreeNode *node)
    {
        TreeNode *succ = node->right;
        while (succ->left != nullptr && succ->left != node)
        {
            succ = succ->left;
        }
        return succ;
    }

    TreeNode *convertBST(TreeNode *root)
    {
        int sum = 0;
        TreeNode *node = root;

        while (node != nullptr)
        {
            if (node->right == nullptr)
            {
                sum += node->val;
                node->val = sum;
                node = node->left;
            }
            else
            {
                TreeNode *succ = getSuccessor(node);
                if (succ->left == nullptr)
                {
                    succ->left = node;
                    node = node->right;
                }
                else
                {
                    succ->left = nullptr;
                    sum += node->val;
                    node->val = sum;
                    node = node->left;
                }
            }
        }

        return root;
    }
};
